#include <stdio.h>
#include <ucontext.h>
#include <unistd.h>
#include <stdlib.h>

static ucontext_t ctx_main,ctx_func1,ctx_func2;

#define handle_error(msg) \
    do { perror(msg); exit(EXIT_FAILURE); } while (0)
void func1(){
    printf("func1 start...\n");
    printf("func1: swapcontext(&uctx_func1, &uctx_func2)\n");
    if(swapcontext(&ctx_func1,&ctx_func2) == -1)
        handle_error("swapcontext");
    printf("func1 returning\n");
}

void func2(){
    printf("func2 start...\n");
    printf("func2: swapcontext(&uctx_func2, &uctx_func1)\n");
    if(swapcontext(&ctx_func2,&ctx_func1) == -1)
        handle_error("swapcontext");
    printf("func2 returning\n");
}

int main(int argc,char** argv){
    char func1_stack[16384];
    char func2_stack[16384];
    if(getcontext(&ctx_func1)==-1)
        handle_error("getcontext");
    ctx_func1.uc_stack.ss_sp = func1_stack;
    ctx_func1.uc_stack.ss_size = sizeof(func1_stack);
    ctx_func1.uc_link = &ctx_main;
    makecontext(&ctx_func1,func1,0);

    if(getcontext(&ctx_func2)==-1)
        handle_error("getcontext");
    ctx_func2.uc_stack.ss_sp = func2_stack;
    ctx_func2.uc_stack.ss_size = sizeof(func2_stack);
    ctx_func2.uc_link = (argc > 1)?NULL:&ctx_func1;
    makecontext(&ctx_func2,func2,0);

    printf("main: swapcontext(&uctx_main, &uctx_func2)\n");
    if(swapcontext(&ctx_main,&ctx_func2)==-1)
        handle_error("swapcontext");
    printf("main exiting...\n");
    return 0;
}